# 修复 PyTorch TorchDynamo 循环导入问题的代码
# 将以下代码添加到你的 Jupyter notebook 的最开始，在导入 torch 之前

import os
# 方法1: 设置环境变量禁用 TorchDynamo
os.environ['TORCH_DYNAMO_DISABLE'] = '1'

# 导入 torch
import torch
print(f"PyTorch 版本: {torch.__version__}")

# 方法2: 在代码中禁用 TorchDynamo
try:
    if hasattr(torch, '_dynamo'):
        torch._dynamo.config.suppress_errors = True
        torch._dynamo.config.disable = True
        print("✓ 已禁用 TorchDynamo")
except Exception as e:
    print(f"⚠ 设置 TorchDynamo 配置时出现警告: {e}")

# 现在可以正常导入其他 PyTorch 模块
import torch.nn as nn
import torch.optim as optim

print("✓ PyTorch 导入成功，可以正常使用了！") 